package stack_queue.leetcode;

import java.util.ArrayList;
import java.util.List;

/**
 * 杨辉三角
 */
public class Num118 {
    public List<List<Integer>> generate(int numRows) {
        List<List<Integer>> retList = new ArrayList<>();
        // 组装特殊的第一行和第二行
        List<Integer> first = new ArrayList<>();
        first.add(1);
        retList.add(first);
        if (numRows == 1) {
            return retList;
        }
        // 组装第二行 [1,1]
        List<Integer> sec = new ArrayList<>();
        sec.add(1);
        sec.add(1);
        retList.add(sec);
        if (numRows == 2) {
            return retList;
        }
        // 此时numRows至少是3行
        for (int i = 3; i <= numRows; i++) {
            // [i,j] = [i - 1,j - 1] + [i - 1,j]
            // 取得当前位置的前一行元素
            List<Integer> prev = retList.get(i - 1 - 1);
            List<Integer> cur = new ArrayList<>();
            // 每一行的第一个元素都是1
            cur.add(1);
            for (int j = 2; j < i; j++) {
                // [i,j] = [i - 1,j - 1] + [i - 1,j]
                int tmpValue = prev.get(j - 1 - 1) + prev.get(j - 1);
                cur.add(tmpValue);
            }
            // 每一行的最后一个元素都是1
            cur.add(1);
            retList.add(cur);
        }
        return retList;
    }
}
